// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.
//
// Automatically generated by addcopyright.py at 01/29/2013
package com.cloud.baremetal.manager;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import javax.inject.Inject;
import javax.naming.ConfigurationException;

import com.cloud.utils.db.QueryBuilder;
import com.cloud.utils.db.SearchCriteria;
import com.cloud.utils.exception.CloudRuntimeException;
import com.cloud.utils.fsm.StateMachine2;
import com.cloud.vm.VMInstanceVO;
import com.cloud.vm.dao.VMInstanceDao;
import org.apache.cloudstack.api.BaremetalProvisionDoneNotificationCmd;

import org.apache.cloudstack.api.AddBaremetalHostCmd;

import com.cloud.host.HostVO;
import com.cloud.host.dao.HostDao;
import com.cloud.hypervisor.Hypervisor.HypervisorType;
import com.cloud.utils.component.ManagerBase;
import com.cloud.utils.fsm.StateListener;
import com.cloud.vm.VirtualMachine;
import com.cloud.vm.VirtualMachine.Event;
import com.cloud.vm.VirtualMachine.State;

public class BaremetalManagerImpl extends ManagerBase implements BaremetalManager, StateListener<State, VirtualMachine.Event, VirtualMachine> {

    @Inject
    protected HostDao _hostDao;
    @Inject
    protected VMInstanceDao vmDao;

    @Override
    public boolean configure(String name, Map<String, Object> params) throws ConfigurationException {
        VirtualMachine.State.getStateMachine().registerListener(this);
        return true;
    }

    @Override
    public boolean start() {
        return true;
    }

    @Override
    public boolean stop() {
        return true;
    }

    @Override
    public String getName() {
        return "Baremetal Manager";
    }

    @Override
    public boolean preStateTransitionEvent(State oldState, Event event, State newState, VirtualMachine vo, boolean status, Object opaque) {
        return false;
    }

    @Override
    public boolean postStateTransitionEvent(StateMachine2.Transition<State, Event> transition, VirtualMachine vo, boolean status, Object opaque) {
      State newState = transition.getToState();
      State oldState = transition.getCurrentState();
      if (newState != State.Starting && newState != State.Error && newState != State.Expunging) {
        return true;
      }

      if (vo.getHypervisorType() != HypervisorType.BareMetal) {
        return true;
      }

      HostVO host = _hostDao.findById(vo.getHostId());
      if (host == null) {
        logger.debug("Skip oldState " + oldState + " to " + "newState " + newState + " transimtion");
        return true;
      }
      _hostDao.loadDetails(host);

      if (newState == State.Starting) {
        host.setDetail("vmName", vo.getInstanceName());
        logger.debug(String.format("Add vmName %s to host %s details", host.getDetail("vmName"), host));
      } else {
        if (host.getDetail("vmName") != null && host.getDetail("vmName").equalsIgnoreCase(vo.getInstanceName())) {
          logger.debug(String.format("Remove vmName %s from host %s details", host.getDetail("vmName"), host));
          host.getDetails().remove("vmName");
        }
      }
      _hostDao.saveDetails(host);

      return true;
    }

  @Override
    public List<Class<?>> getCommands() {
        List<Class<?>> cmds = new ArrayList<Class<?>>();
        cmds.add(AddBaremetalHostCmd.class);
        cmds.add(BaremetalProvisionDoneNotificationCmd.class);
        return cmds;
    }

    @Override
    public void notifyProvisionDone(BaremetalProvisionDoneNotificationCmd cmd) {
        QueryBuilder<HostVO> hq = QueryBuilder.create(HostVO.class);
        hq.and(hq.entity().getPrivateMacAddress(), SearchCriteria.Op.EQ, cmd.getMac());
        HostVO host = hq.find();
        if (host == null) {
            throw new CloudRuntimeException(String.format("cannot find host[mac:%s]", cmd.getMac()));
        }

        _hostDao.loadDetails(host);
        String vmName = host.getDetail("vmName");
        if (vmName == null) {
            throw new CloudRuntimeException(String.format("cannot find any baremetal instance running on host[mac:%s]", cmd.getMac()));
        }

        QueryBuilder<VMInstanceVO> vmq = QueryBuilder.create(VMInstanceVO.class);
        vmq.and(vmq.entity().getInstanceName(), SearchCriteria.Op.EQ, vmName);
        VMInstanceVO vm = vmq.find();

        if (vm == null) {
            throw new CloudRuntimeException(String.format("cannot find baremetal instance[name:%s]", vmName));
        }

        if (State.Starting != vm.getState()) {
            throw new CloudRuntimeException(String.format("baremetal instance %s [state:%s] is not in state of Starting", vm, vm.getState()));
        }

        vm.setState(State.Running);
        vm.setLastHostId(vm.getHostId());
        vmDao.update(vm.getId(), vm);
        logger.debug(String.format("received baremetal provision done notification for vm %s running on host %s [mac:%s, ip:%s]",
                vm, host, host.getPrivateMacAddress(), host.getPrivateIpAddress()));
    }
}
